import torch

GPU_id = torch.cuda.current_device()
print(f'{type(GPU_id)}|{GPU_id}|')
